package com.tianji.learning.utils;

import lombok.Data;

import java.time.Duration;
import java.util.concurrent.Delayed;
import java.util.concurrent.TimeUnit;

// 定义一个泛型类 DelayTask，实现 Delayed 接口
@Data
public class DelayTask<D> implements Delayed {
    // 存储任务数据
    private D data;
    // 任务截止时间，以纳秒为单位
    private long deadlineNanos;

    /**
     * 构造函数，初始化任务数据和截止时间
     * @param data 任务数据
     * @param delayTime 延迟时间
     */
    public DelayTask(D data, Duration delayTime) {
        this.data = data;
        // 计算截止时间，当前时间加上延迟时间
        this.deadlineNanos = System.nanoTime() + delayTime.toNanos();
    }

    /**
     * 获取任务剩余延迟时间
     * @param unit 时间单位
     * @return 剩余延迟时间
     */
    @Override
    public long getDelay(TimeUnit unit) {
        // 计算剩余延迟时间，返回指定时间单位的值
        return unit.convert(Math.max(0, deadlineNanos - System.nanoTime()), TimeUnit.NANOSECONDS);
    }

    /**
     * 比较两个任务的延迟时间
     * @param o 另一个任务
     * @return 比较结果
     */
    @Override
    public int compareTo(Delayed o) {
        // 计算两个任务的延迟时间差
        long l = getDelay(TimeUnit.NANOSECONDS) - o.getDelay(TimeUnit.NANOSECONDS);
        if (l > 0) {
            // 当前任务延迟时间更长，返回1
            return 1;
        } else if (l < 0) {
            // 当前任务延迟时间更短，返回-1
            return -1;
        } else {
            // 两个任务延迟时间相同，返回0
            return 0;
        }
    }
}
